/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */

package main.initiate;

/**
 *
 * @author Isaac Osesina
 */

import java.io.File;
import java.io.Serializable;
import java.util.Set;
import main.Ontology.DefaultElementOntologyFactory;
import main.Ontology.ElementOntologyFactory;

public class CreateKBIndex implements Cloneable, Serializable{
        File kbFile;
        File ontologyFile;
        Set<Integer> docsToAccept, docsToSkip;
        UTEES_Index kbIndex;

    public CreateKBIndex(String kbFileName, String ontologyFileName, Set<Integer> docsToSkip, Set<Integer> docsToAccept) {
        this.docsToSkip = docsToSkip;
        this.docsToAccept = docsToAccept;

        ElementOntologyFactory f = new DefaultElementOntologyFactory();
        kbIndex = new UTEES_Index(kbFileName, ontologyFileName, docsToSkip, docsToAccept, f);
    }
    
    public CreateKBIndex(Set<Integer> docsToSkip, Set<Integer> docsToAccept) {
        this.docsToSkip = docsToSkip;
        this.docsToAccept = docsToAccept;

        ElementOntologyFactory f = new DefaultElementOntologyFactory();
        kbIndex = new UTEES_Index(docsToSkip, docsToAccept, f);
    }

    public UTEES_Index getKbIndex() {
        return kbIndex;
    }

}
